package cn.wolfcode.trip.base.util;

import cn.wolfcode.trip.base.domain.User;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpSession;
import java.util.Set;

//用户操作相关的工具类
public class UserContext {
    public static final String USER_IN_SESSION = "USER_IN_SESSION";

    private UserContext() {
    }

    public static HttpSession getSession() {
        RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) attributes;
        return requestAttributes.getRequest().getSession();
    }

    public static void setCurrentUser(User user) {
        getSession().setAttribute(USER_IN_SESSION, user);
    }


    public static User getCurrentUser() {
        return (User) getSession().getAttribute(USER_IN_SESSION);
    }


}
